use crate::Tensor;

pub trait Module<T> {
    fn forward(&self, tensor: Tensor<T>) -> Tensor<T>;
    fn parameters(&self) -> Vec<Tensor<T>>;
}
